package fdupc.crypto.aes;

import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;

/**
 * 名称: AES处理基础抽象
 *
 * @description:
 * @author: XXSD
 * @date: 2023-06-30 20:08
 */
public class AbstractAesBase implements IAesManager {

    /**
     * 功能描述：加密处理
     *
     * @param srcStr 明文
     * @param srcKey 密钥
     * @param iv     向量
     * @date: 2023/1/31 0031 下午 5:42
     */
    public String encrypt(String srcStr, String srcKey, String iv) throws Exception {
        return doEncrypt(srcStr, StandardCharsets.UTF_8, srcKey, iv);
    }

    /**
     * 功能描述：加密处理
     *
     * @param srcStr  明文
     * @param charset 字符集，默认utf8
     * @param srcKey  密钥
     * @param iv      向量
     * @date: 2023/1/31 0031 下午 5:42
     */
    public String encrypt(String srcStr, Charset charset, String srcKey, String iv) throws Exception {
        return doEncrypt(srcStr, (charset == null) ? StandardCharsets.UTF_8 : charset, srcKey, iv);
    }

    /**
     * 解密
     *
     * @param srcStr 密文
     * @param srcKey 密钥
     * @param iv     向量
     * @return 明文
     */
    public String decrypt(String srcStr, String srcKey, String iv) throws Exception {
        return doDecrypt(srcStr, StandardCharsets.UTF_8, srcKey, iv);
    }

    /**
     * 解密
     *
     * @param srcStr  密文
     * @param srcKey  密钥
     * @param charset 字符集，默认utf8
     * @param iv      向量
     * @return 明文
     */
    public String decrypt(String srcStr, Charset charset, String srcKey, String iv) throws Exception {
        return doDecrypt(srcStr, (charset == null) ? StandardCharsets.UTF_8 : charset, srcKey, iv);
    }

    /**
     * 功能描述：加密处理
     *
     * @param srcStr  明文
     * @param charset 字符集，默认utf8
     * @param srcKey  密钥
     * @param iv      向量
     * @date: 2023/1/31 0031 下午 5:42
     */
    protected String doEncrypt(String srcStr, Charset charset, String srcKey, String iv) throws Exception {
        return dataEncrypt(giveCipher(), decodeBase64(srcStr), charset, srcKey, iv);
    }

    /**
     * 功能名称:
     *
     * @param cipher  加密处理对象
     * @param srcStr  明文
     * @param charset 字符集，默认utf8
     * @param srcKey  密钥
     * @param iv      向量
     * @return 加密结果
     * @description:
     * @author: XXSD
     * @date: 2023-06-30 21:11
     */
    protected String dataEncrypt(Cipher cipher, byte[] srcStr, Charset charset, String srcKey, String iv) throws Exception {
        SecretKeySpec keyspec = new SecretKeySpec(srcKey.getBytes(charset), "AES");
        // CBC模式，需要一个向量iv，可增加加密算法的强度
        IvParameterSpec ivspec = new IvParameterSpec(iv.getBytes(charset));

        cipher.init(Cipher.ENCRYPT_MODE, keyspec, ivspec);
        byte[] encrypted = cipher.doFinal(srcStr);
        // BASE64做转码。
        return encodeBase64(encrypted).trim();
    }

    /**
     * 解密
     *
     * @param srcStr  密文
     * @param charset 字符集，默认utf8
     * @param srcKey  密钥
     * @param iv      向量
     * @return 明文
     */
    protected String doDecrypt(String srcStr, Charset charset, String srcKey, String iv) throws Exception {
        return dataDecrypt(giveCipher(), decodeBase64(srcStr), charset, srcKey, iv);
    }

    /**
     * 解密
     *
     * @param srcStr  密文
     * @param charset 字符集，默认utf8
     * @param srcKey  密钥
     * @param iv      向量
     * @return 明文
     */
    protected String dataDecrypt(Cipher cipher, byte[] srcStr, Charset charset, String srcKey, String iv) throws Exception {
        SecretKeySpec secretKey = new SecretKeySpec(srcKey.getBytes(charset), "AES");
        IvParameterSpec ivParameterSpec = new IvParameterSpec(iv.getBytes(charset));
        cipher.init(Cipher.DECRYPT_MODE, secretKey, ivParameterSpec);
        return new String(cipher.doFinal(srcStr), charset).trim();
    }
}
